class Solution 
{
public:
    void setZeroes(vector<vector<int>>& matrix) 
    {
        int row = matrix.size(), col = matrix[0].size();;
        bool row0_flag = false, col0_flag = false;
        for (int j = 0; j < col; ++j) // 检查第一行是否有0
        {
            if (matrix[0][j] == 0) 
            {
                row0_flag = true;
                break;
            }
        }
        for (int i = 0; i < row; ++i) // 检查第一列是否有0
        {
            if (matrix[i][0] == 0) 
            {
                col0_flag = true;
                break;
            }
        }
        for (int i = 1; i < row; ++i) // 使用第一行和第一列作为标志位
        {
            for (int j = 1; j < col; ++j) 
            {
                if (matrix[i][j] == 0) 
                {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        for (int i = 1; i < row; ++i)  // 根据标志位置0
        {
            for (int j = 1; j < col; ++j) 
            {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) 
                {
                    matrix[i][j] = 0;
                }
            }
        }
        if (row0_flag) // 处理第一行
        {
            for (int j = 0; j < col; ++j) 
            {
                matrix[0][j] = 0;
            }
        }
        if (col0_flag)  // 处理第一列
        {
            for (int i = 0; i < row; ++i) 
            {
                matrix[i][0] = 0;
            }
        }
    }
};